"""
日志配置模块
"""
import os
from loguru import logger

from config.settings import LOG_DIR

def setup_logger(level="INFO"):
    """
    配置日志记录器
    
    Args:
        level: 日志级别，默认为INFO
    """
    # 移除默认的处理器
    logger.remove()
    
    # 添加控制台处理器
    logger.add(
        sink=lambda msg: print(msg),
        level=level,
        format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}"
    )
    
    # 确保日志目录存在
    os.makedirs(LOG_DIR, exist_ok=True)
    
    # 添加文件处理器
    log_file = os.path.join(LOG_DIR, "stock_data_{time:YYYYMMDD}.log")
    logger.add(
        sink=log_file,
        level=level,
        format="{time:YYYY-MM-DD HH:mm:ss} | {level} | {message}",
        rotation="00:00",  # 每天轮换
        retention="30 days"  # 保留30天
    ) 